package com.ge.myoned.util;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.core.toolkit.ArrayUtils;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert;
import com.baomidou.mybatisplus.generator.config.po.TableField;
import com.baomidou.mybatisplus.generator.config.po.TableFill;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.IColumnType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.VelocityTemplateEngine;

import java.util.ArrayList;
/**
 * @Author geqingxiang
 * @PackageName my-project-test
 * @Package com.ge
 * @Date 2023/9/15 14:31
 * @Version 1.0
 */
public class MybatisPlusGenerator {

    //生成文件所在项目路径
    private static String baseProjectPath = "D:\\generatorCode";

    //基本包名
    private static String basePackage = "com.tunsda.openai";

    //作者(一般公司名简称)
    private static String authorName = "tunda";

    //要生成的表名，为空时生成所有表
    private static String[] tables = {"t_goods_info"};

    //table前缀:生成代码的时候可以忽略前缀，生成实体类：
    /**例如: private static String prefix = "t_";
     则t_goods_info的表，生成实体类则是GoodsInfo而不是TGoodsInfo
     */
    private static String prefix = "";

    //数据库配置四要素
    private static String driverName = "com.mysql.jdbc.Driver";
    private static String url = "jdbc:mysql://127.158.254.38:3306/tusnsda_gpt?serverTimezone=GMT%2B8&zeroDateTimeBehavior=CONVERT_TO_NULL&useUnicode=true&characterEncoding=utf-8&useSSL=false";
    private static String username = "root";
    private static String password = "root";

    public static void main(String[] args) {
        // 代码生成器
        AutoGenerator mpg = new AutoGenerator();

        // 1、全局配置
        GlobalConfig gc = new GlobalConfig();

        //可以将生成得代码单独放在文件夹里，检查完没问题在手动将包复制到项目中
        gc.setOutputDir(baseProjectPath);
        gc.setAuthor(authorName);// 生成实体类的文档注解中作者名
        gc.setOpen(true);// 生成代码后自动打开生成目录文件
        gc.setFileOverride(true); // 是否覆盖原来生成的
        gc.setIdType(IdType.AUTO);// 指定生成的主键类型：自增长
        gc.setDateType(DateType.ONLY_DATE);
        gc.setActiveRecord(false);
        mpg.setGlobalConfig(gc);

        // 2、数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl(url);
        dsc.setDriverName(driverName);
        dsc.setUsername(username);
        dsc.setPassword(password);
        dsc.setDbType(DbType.MYSQL); // mysql数据库
        gc.setServiceName("%sService"); //去掉Service接口的首字母'I'
        dsc.setTypeConvert(new MySqlTypeConvert() {
            @Override
            public IColumnType processTypeConvert(GlobalConfig globalConfig, TableField tableField) {
                return super.processTypeConvert(globalConfig, tableField);
            }
        });
        mpg.setDataSource(dsc);

        // 3、包配置
        PackageConfig pc = new PackageConfig();
        pc.setModuleName(null)// 生成的controller路径映射为一个"/",覆盖默认的"//"
                .setParent(basePackage)// 生成controller，service，xml并设置路径。
                .setEntity("entity")
                .setMapper("mapper")
                .setService("service")
                .setServiceImpl("service.impl")
                .setController("controller");
        mpg.setPackageInfo(pc);

        // 5、策略配置
        StrategyConfig strategy = new StrategyConfig();
        strategy.setCapitalMode(true);// 开启全局大写命名
        strategy.setNaming(NamingStrategy.underline_to_camel);// 实体类名驼峰
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);// 属性名驼峰
        strategy.setInclude(tables);// 设置要映射的表名注解@Table
//        strategy.setSuperEntityClass(BaseEntity.class); // 预先定义一个BaseEntity类实现Serializable接口
        strategy.setEntityLombokModel(true); // 使用 lombok
        strategy.setRestControllerStyle(true);// 开启 restful风格
        strategy.setEntityTableFieldAnnotationEnable(false);// 不生成实体类字段swagger类型注解,而是采用文档类型注解：/**xxx*/

        if (prefix != null) strategy.setTablePrefix(prefix);  // 表前缀
        if (ArrayUtils.isNotEmpty(tables)) strategy.setInclude(tables);  // 表名

        // 6、自动填充配置：非必须
        TableFill createTime = new TableFill("create_time", FieldFill.INSERT);
        TableFill updateTime = new TableFill("update_time", FieldFill.INSERT_UPDATE);
        ArrayList<TableFill> tableFills = new ArrayList<>();
        tableFills.add(createTime);
        tableFills.add(updateTime);
        strategy.setTableFillList(tableFills);
        strategy.setLogicDeleteFieldName("deleted");// 指定逻辑删除字段

        mpg.setStrategy(strategy);

        mpg.setTemplateEngine(new VelocityTemplateEngine());

        // 执行生成器
        mpg.execute();
    }

}
